#!/bin/bash
module purge
module load anaconda cuda/11.1 complier/gcc/12.2.0 openmpi/4.1.1
source activate bevfusion
export PYTHONPATH=$PWD
export CUDA_HOME='/data/apps/cuda/11.1'
torchpack dist-run -np 8 python tools/train.py --config_s configs/nuscenes/det/centerhead/lssfpn/camera/256x704/swint/default.yaml   --config_t configs/nuscenes/det/centerhead/lssfpn/camera/256x704/swint/default.yaml   --checkpoint pretrained/camera-only-det.pth  --model.encoders.camera.backbone.init_cfg.checkpoint pretrained/swint-nuimages-pretrained.pth
#python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes
